package com.sgj.sparsearray;

/**
 * 稀疏数组与二维数组的转换
 * @Author: Guoji Shen
 * @Date: 2021/7/30 14:15
 */
public class SparseArray {
    public static void main(String[] args) {
        // 创建一个原始的二维数组  11 * 11
        int chessArr[][] = new int[11][11];
        chessArr[1][2] = 1;
        chessArr[2][3] = 2;
        chessArr[3][4] = 3;

        // 输出原始的二维数组
        System.out.println("原始的二维数组为：");
        original(chessArr);

        // 将二维数组 转为 稀疏数组
        System.out.println("\n将二维数组转为稀疏数组为：");
        int[][] sparsearray = twoDimensionalToSparsearray(chessArr);

        // 将稀疏数组恢复成原始的二维数组
        System.out.println("\n稀疏数组恢复成原始的二维数组为：");
        sparsearrayToTwoDimensional(sparsearray);
    }

    /**
     * 二维数组遍历
     * @param chessArr  二维数组
     */
    public static void original (int chessArr[][]) {
        for (int[] row : chessArr) {
            for (int data : row) {
                // 格式化输出
                System.out.printf("%d\t", data);
            }
            System.out.println();
        }
    }

    /**
     * 将二位数组转换为稀疏数组
     * @param chessArr    原始的二维数组
     * @return sparseArr  返回的稀疏数组
     */
    public static int[][] twoDimensionalToSparsearray (int chessArr[][]) {
        // 先遍历二维数组，得到非0数据的个数
        int sum = 0;
        for (int i = 0; i < 11; i++) {
            for (int j = 0; j < 11; j++) {
                if (chessArr[i][j] != 0) {
                    sum ++;
                }
            }
        }

        // 创建对应的稀疏数组
        int sparseArr[][] = new int[sum + 1][3];
        // 给稀疏数组赋值
        sparseArr[0][0] = 11;
        sparseArr[0][1] = 11;
        sparseArr[0][2] = sum;

        // 遍历二维数组，将非0的值存放到稀疏数组中
        // 记录是第几个的非0数据
        int count = 0;
        for (int i = 0; i < 11; i++) {
            for (int j = 0; j < 11; j++) {
                if (chessArr[i][j] != 0) {
                    count ++;
                    sparseArr[count][0] = i;
                    sparseArr[count][1] = j;
                    sparseArr[count][2] = chessArr[i][j];
                }
            }
        }

        // 输出稀疏数组
        for (int i = 0; i < sparseArr.length; i++) {
            System.out.printf("%d\t%d\t%d\t\n", sparseArr[i][0], sparseArr[i][1], sparseArr[i][2]);
        }
        return sparseArr;
    }

    /**
     * 稀疏数组转换为二维数组
     * @param sparsearray  稀疏数组
     */
    public static void sparsearrayToTwoDimensional (int sparsearray[][]) {
        // 先读取稀疏数组的第一行，根据第一行的数据，还原原始的二维数组
        int chessArr[][] = new int[sparsearray[0][0]][sparsearray[0][1]];

        // 遍历稀疏数组从第二行开始的数据，因为第一行是原数组的格式，第二行开始才是数据的存放
        for (int i = 1; i < sparsearray.length; i++) {
            chessArr[sparsearray[i][0]][sparsearray[i][1]] = sparsearray[i][2];
        }
        original(chessArr);
    }
}